package com.liuyjy.base.common.util;

import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.liuyjy.base.common.entity.PageResult;

import java.util.ArrayList;
import java.util.List;

/**
 * @Author liuyjy
 * @Date 2023/1/31 14:35
 * @Description: 分页方法
 **/
public class PagesUtil {

    /**
     * 分页函数
     *
     * @param currentPage 当前页数
     * @param pageSize    每一页的数据条数
     * @param list        要进行分页的数据列表
     * @return 当前页要展示的数据
     */
    public static <T> PageResult<T> getPageResult(Integer currentPage, Integer pageSize, List<T> list) {
        return ObjConvertUtil.convert(getPages(currentPage, pageSize, list));
    }

    /**
     * 分页函数
     *
     * @param currentPage 当前页数
     * @param pageSize    每一页的数据条数
     * @param list        要进行分页的数据列表
     * @return 当前页要展示的数据
     */
    public static <T> Page<T> getPages(Integer currentPage, Integer pageSize, List<T> list) {
        Page<T> page = new Page<T>(currentPage, pageSize);
        if (list == null) {
            page.setRecords(new ArrayList<T>());
            page.setTotal(0);
            return page;
        }
        int size = list.size();

        if (pageSize > size) {
            pageSize = size;
        }
        if (pageSize != 0) {
            // 求出最大页数，防止currentPage越界
            int maxPage = size % pageSize == 0 ? size / pageSize : size / pageSize + 1;

            if (currentPage > maxPage) {
                currentPage = maxPage;
            }
        }
        // 当前页第一条数据的下标
        int curIdx = currentPage > 1 ? (currentPage - 1) * pageSize : 0;

        List<T> pageList = new ArrayList<T>();

        // 将当前页的数据放进pageList
        for (int i = 0; i < pageSize && curIdx + i < size; i++) {
            pageList.add(list.get(curIdx + i));
        }

        page.setTotal(list.size()).setRecords(pageList);
        return page;
    }

}
